import pandas as pd

# 读取数据
data_path = 'D:\\学习&科研\\华为手表项目\\华为数据\\试验记录表\\all_stages_df.csv'
data = pd.read_csv(data_path)
info = pd.read_csv('D:\\学习&科研\\华为手表项目\\华为数据\\试验记录表\\受试者信息统计表utf8.csv')

# 替换性别信息
info['sex'] = info['sex'].replace({'男': 1, '女': 0})

# 提取 data 中 number 的前四位
data['id'] = data['number'].astype(str).str[:4]

# 确保 info 中的 id 列为字符串类型
info['id'] = info['id'].astype(str)

# 使用 merge 函数将 info 的所有列填充到 data 中
data = data.merge(info, on='id', how='left')

# 显示结果
print(info)
print(data)
data.to_csv('D:\\学习&科研\\华为手表项目\\华为数据\\试验记录表\\all_stages_df_info.csv', index=False)
